import  requests
sess=requests.session()

IP= '192.168.1.180'

# 银行登录接口（cookie）	POST	/pinter/bank/api/login	userName=admin&password=1234

def cookie_admin_test():
    data={"userName":"admin","password":"1234"}
    rp=sess.request('post', url='http://' + IP + ':8080/pinter/bank/api/login', data=data)
    print(rp.text)


# 账户余额查询接口（cookie）   get	/pinter/bank/api/query	userName=admin
def cookie_test():
    # 1. 登录后获得权利
    data = {"userName": "admin", "password": "1234"}
    rp = sess.request('post', url='http://' + IP + ':8080/pinter/bank/api/login', data=data)
    # 2. 鉴定它有没有权利
    params = {"userName": "admin"}
    rp = sess.request('get', url='http://' + IP + ':8080/pinter/bank/api/query', params=params)
    print(rp.text)

# 账户余额查询接口（token）       get	/pinter/bank/api/query	userName=admin
def token_test():
    # 1. 登录后获得权利
    data={"userName":"admin","password":"1234"}
    rp=sess.request('post', url='http://' + IP + ':8080/pinter/bank/api/login', data=data)
    mytoken =rp.json().get("data") #鉴定接口的请求头，添加一个属性

    # 2. 鉴定它有没有权利
    params={"userName":"admin"}
    headers ={"testfan-token":mytoken}   #提取接口的请求头的属性
    rp=sess.request('get', url='http://' + IP + ':8080/pinter/bank/api/query', params=params, headers=headers)
    print(rp.text)



if __name__=='__main__':
    cookie_admin_test()
    cookie_test()
    token_test()
